package com.git.bcs.common.bean;

import java.sql.SQLException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;
import com.git.bcs.common.bean.properties.DruidProperties;

@Configuration
public class DataSourceDruidConf {

	private static final Logger logger = LoggerFactory.getLogger(DataSourceDruidConf.class);

	private static final String DRUIDUSERNAME = "admin";
	private static final String DRUIDPASSWORD = "admin";

	@Autowired
	DruidProperties druidProperties;

	@Bean(initMethod = "init",name="datasource")
	@Primary
	public DruidDataSource druidDataSource() throws SQLException {

		DruidDataSource dataSource = new DruidDataSource();
		dataSource.setUrl(druidProperties.getUrl());
		dataSource.setDriverClassName(druidProperties.getDriverclassname());
		dataSource.setUsername(druidProperties.getUsername());
		dataSource.setPassword(druidProperties.getPassword());
		dataSource.setInitialSize(Integer.valueOf(druidProperties.getInitialSize()));
		dataSource.setMaxActive(Integer.valueOf(druidProperties.getMaxActive()));
		dataSource.setMinIdle(Integer.valueOf(druidProperties.getMinIdle()));
		dataSource.setMaxWait(Integer.valueOf(druidProperties.getMaxWait()));
		
		dataSource
				.setTimeBetweenEvictionRunsMillis(Integer.valueOf(druidProperties.getTimeBetweenEvictionRunsMillis()));
		dataSource.setMinEvictableIdleTimeMillis(Integer.valueOf(druidProperties.getMinEvictableIdleTimeMillis()));
		dataSource.setValidationQuery(druidProperties.getValidationQuery());
		dataSource.setTestWhileIdle(Boolean.getBoolean(druidProperties.getTestWhileIdle()));
		dataSource.setTestOnBorrow(Boolean.getBoolean(druidProperties.getTestOnBorrow()));
		dataSource.setTestOnReturn(Boolean.getBoolean(druidProperties.getTestOnReturn()));
		dataSource.setPoolPreparedStatements(Boolean.getBoolean(druidProperties.getPoolPreparedStatements()));
		dataSource.setMaxPoolPreparedStatementPerConnectionSize(
				Integer.valueOf(druidProperties.getMaxPoolPreparedStatementPerConnectionSize()));

		// dataSource.setConnectionProperties();

		dataSource.setInitialSize(10);
		try {
			dataSource.setFilters(druidProperties.getFilters());
		} catch (SQLException e) {
			e.printStackTrace();
		}
	

		return dataSource;
	}

	/**
	 * 注册一个StatViewServlet
	 * 
	 * @return
	 */
	@Bean
	public ServletRegistrationBean DruidStatViewServle2() {
		// org.springframework.boot.context.embedded.ServletRegistrationBean提供类的进行注册.
		ServletRegistrationBean servletRegistrationBean = new ServletRegistrationBean(new StatViewServlet(),
				"/druid/*");
		// 添加初始化参数：initParams
		// 白名单：
		servletRegistrationBean.addInitParameter("allow", "127.0.0.1");
		// IP黑名单 (存在共同时，deny优先于allow) : 如果满足deny的话提示:Sorry, you are not permitted to
		// view this page.
		/* servletRegistrationBean.addInitParameter("deny","192.168.1.73"); */
		// 登录查看信息的账号密码.
		servletRegistrationBean.addInitParameter("loginUsername", DRUIDUSERNAME);
		servletRegistrationBean.addInitParameter("loginPassword", DRUIDPASSWORD);
		// 是否能够重置数据.
		servletRegistrationBean.addInitParameter("resetEnable", "false");
		return servletRegistrationBean;

	}

	/**
	 * 注册一个：filterRegistrationBean
	 * 
	 * @return
	 */

	@Bean
	public FilterRegistrationBean druidStatFilter2() {
		FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean(new WebStatFilter());
		// 添加过滤规则.
		filterRegistrationBean.addUrlPatterns("/*");
		// 添加不需要忽略的格式信息.
		filterRegistrationBean.addInitParameter("exclusions", "*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
		return filterRegistrationBean;

	}
}
